[2주차 Stack & Hash] 방문길이 - Lv3

문제 설명

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.

U: 위쪽으로 한 칸 가기
D: 아래쪽으로 한 칸 가기
R: 오른쪽으로 한 칸 가기
L: 왼쪽으로 한 칸 가기

캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

예를 들어, ULURRDLLU로 명령했다면

1번 명령어부터 7번 명령어까지 다음과 같이 움직입니다.

8번 명령어부터 9번 명령어까지 다음과 같이 움직입니다.

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)

단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.

예를 들어, LULLLLLLU로 명령했다면

1번 명령어부터 6번 명령어대로 움직인 후, 7, 8번 명령어는 무시합니다. 다시 9번 명령어대로 움직입니다.

이때 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다.

명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여 return 하는 solution 함수를 완성해 주세요.

제한사항

dirs는 string형으로 주어지며, ‘U’, ‘D’, ‘R’, ‘L’ 이외에 문자는 주어지지 않습니다.
dirs의 길이는 500 이하의 자연수입니다.

입출력 예

dirs	answer
ULURRDLLU	7
LULLLLLLU	7

입출력 예 설명

입출력 예 #1 문제의 예시와 같습니다.

입출력 예 #2 문제의 예시와 같습니다.

나의 풀이

첫번째 풀이

소스

def solution(dirs):
    answer = 0
    cur_position = [0, 0]
    visit_logs = set()

    for dir in dirs:
        cur_xpos, cur_ypos = cur_position
        # 명령어를 체크하고 다음위치를 확인한다.
        if dir == "U":
            next_position = [cur_xpos, cur_ypos + 1]
        elif dir == "L":
            next_position = [cur_xpos - 1, cur_ypos]
        elif dir == "D":
            next_position = [cur_xpos, cur_ypos - 1]
        elif dir == "R":
            next_position = [cur_xpos + 1, cur_ypos]

        # 벗어나는 영역이면 현재 위치 그대로 두고 넘어간다.
        if next_position[0] > 5 or next_position[0] < -5 or next_position[1] > 5 or next_position[1] < -5:
            continue

        # 0,0-1,0 과 같은 형태로 키를 만든다.
        visit_log = ','.join(map(str, cur_position)) + \
            '-' + ','.join(map(str, next_position))

        visit_log_reverse = ','.join(
            map(str, next_position)) + '-' + ','.join(map(str, cur_position))

        # 방문하지 않았다면 visit_logs에 방문을 기록하고 방문길이를 더한다.
        if visit_log not in visit_logs:
            answer += 1
            visit_logs.add(visit_log)
            visit_logs.add(visit_log_reverse)

        # 다음 위치로 이동한다.
        cur_position = next_position

    return answer

설명

이 문제는 방문 기록을 해쉬를 통해 관리하면 쉽게 풀 수 있다.
하지만 역순으로도 기록을 해야 된다는 점이 문제풀이의 핵심.
각 명령어에 맞게 위치를 조정하고 지나온 길을 해쉬에 기록.
키를 나는 string으로 했는데, unmutable한 tuple도 딕셔너리의 키가 될 수 있었다.
질문
- 이렇게 구현하면 O(n)이겠죠?
- 키 값을 제가 지나온 길을 “0,0-1,0” 과 같은 형태의 string으로 만들어서 넣는데요 (“1,0-0,0” 도 같이) 더 좋은 방법의 키값설정이 있을까요?

결과

통과

리뷰

dirs만큼 순회하기 때문에 O(n)이 맞습니다. :)

string대신 다음과 같이 tuple로 설정할 수 있습니다.
visit_log = (*cur_position, *next_position)
visit_log_reverse = (*next_position, *cur_position)

스터디 리더의 풀이

풀이 1

def solution(dirs):
    x, y = 0, 0 # 시작 좌표를 0, 0으로 지정
    map = dict() # 좌표를 키로 사용하는 해시 생성
    for command in dirs: # O(dirs)
        if command == 'U' and y < 5: # 위로
            map[(x, y, x, y+1)] = True # 현재 좌표, 이동할 좌표
            # x, y 좌표 작은게 왼쪽으로~
            y += 1
        elif command == 'D' and y > -5: # 아래로
            map[(x, y-1, x, y)] = True # 이동할 좌표, 현재 좌표
            y -= 1
        elif command == 'R' and x < 5: # 오른쪽으로
            map[(x, y, x+1, y)] = True # 현재 좌표, 이동할 좌표
            x += 1
        elif command == 'L' and x > -5: # 왼쪽으로
            map[(x-1, y, x, y)] = True # 이동할 좌표, 현재 좌표.
            x -= 1

    return len(map) # 추가된 값들이 곧 방문 길이

풀이2

def solution(dirs):
    position = (0, 0) # 시작 좌표를 0, 0으로 지정

    command_dict = {
        'U': (-1, 0),
        'D': (1, 0),
        'R': (0, 1),
        'L': (0, -1)
    }

    check = set() # 좌표를 키로 사용하는 해시 생성
    for command in dirs: # O(dirs)
        direction = command_dict[command]
        next_position = tuple(map(sum, zip(position, direction)))

        y, x = next_position
        if -5 <= y <= 5 and -5 <= x <= 5:
            check.add(tuple(sorted([position, next_position])))
            position = next_position

    return len(check)